from .PReNet import PReNet

def build_model(args):
    if args.datasets == "prenet":
        model= PReNet()
   else:
        raise NotImplementedError

        
    return model